import torch
from torchinfo import summary
model = torch.load(r'E:\study_code\torch_stud_1\pytorch深度学习应用实战\第4章神经网络实操\model.pt',weights_only=False)
print(summary(model=model, input_size=(6000, 28, 28)))